from langchain.chains import LLMChain
#提示词的模版
from langchain import PromptTemplate
#引入百度千帆平台的大语言模
from langchain_community.llms import Ollama
#llm(large language model) 大语言模型
#ERNIE-Bot 4.0
#Qianfan-Chinese-Llama-2-7B
def get_response_from_llm(question:str):
    llm = Ollama(model='qwen:1.8b')

    template = """问题: {question}
    回答: 请一步一步思考，然后回答."""

    prompt = PromptTemplate(template=template, input_variables=["question"])
    llm_chain = LLMChain(prompt=prompt, llm=llm)

    response = llm_chain.run(question)
    # response = llm.invoke(question)
    return response

result = get_response_from_llm("告诉我如何做鱼香肉丝")
print(result)